package com.atguigu.gulimall.order.config;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.annotation.EnableRabbit;
import org.springframework.amqp.rabbit.connection.ConnectionFactory;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.amqp.support.converter.Jackson2JsonMessageConverter;
import org.springframework.amqp.support.converter.MessageConverter;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;

/**
 * @author: haitao
 * @email: haitaoss@aliyun.com
 * @date: 2020/7/16 17:25:57
 */
@Configuration
@EnableRabbit
public class MyRabbitConfig {
    //    @Autowired
    RabbitTemplate rabbitTemplate;

    // TODO
    @Bean
    @Primary
    public RabbitTemplate rabbitTemplate(ConnectionFactory connectionFactory) {
        RabbitTemplate template = new RabbitTemplate(connectionFactory);
        template.setMessageConverter(Jackson2JsonMessageConverter());
        this.rabbitTemplate = template;
        initRabbitTemplate();
        return template;
    }


    @Bean
    public MessageConverter Jackson2JsonMessageConverter() {
        return new Jackson2JsonMessageConverter();
    }

    /**
     * 定制RabbitTemplate
     * 1、服务端收到消息就回调
     * 1、spring.rabbitmq.publisher-confirms=true
     * 2、设置确认回调ConfirmCallback
     * <p>
     * 2、消息没有正确抵达队列进行回调
     * 1、spring.rabbitmq.publisher-returns=true、spring.rabbitmq.template.mandatory=true
     * 2、设置失败回调ReturnCallback
     * <p>
     * 3、消费端确认（保证每个消息被正确消费，此时才可以broker删除这个消息）
     * spring.rabbitmq.listener.simple.acknowledge-mode=manual 开启手动签收
     * 1、默认是自动确认的，只要消息接受到，消费端端会自动确认，服务端就会移除这个消息。
     * 问题：
     * 我们收到很多消息，自动回复给服务端ack。只有一个消息处理成功，宕机了。发生消息丢失。（通道一打开就回复ack）
     * 手动确认模式。只要我们没有明确告诉MQ，消息就一直是unacked转态。即使consumer宕机。队列里面的消息也不会丢失，consumer宕机则队列里面的消息会变为ready，
     * 下次有新的consumer连接进来就发给他（队列的消息变成unacked）。队列消息的转态是根据有没有消费者连接上队列
     * 2、如何签收：
     * channel.basicAck(deliveryTag, false);签收：业务成功完成就应该签收
     * channel.basicNack(deliveryTag,false,true); 拒签：业务失败，拒签。
     */
//    @PostConstruct // MyRabbitConfig 构造器创建完成以后，执行这个方法
    public void initRabbitTemplate() {
        // 设置确认回调
        this.rabbitTemplate.setConfirmCallback(new RabbitTemplate.ConfirmCallback() {
            /**
             * 1、 只要消息抵达Broker就ack=true
             * @param correlationData 当前消息的唯一关联数据(这个是消息的唯一id)
             * @param ack 消息是否成功收到
             * @param cause 失败的原因
             */
            @Override
            public void confirm(CorrelationData correlationData, boolean ack, String cause) {
                /*
                    1、做好消息确认机制（publisher， consumer 【手动ack】）
                    2、每一个发送的消息都在数据库做好记录。定期将失败的消息再次发送一遍
                 */
                // 服务器收到了
                // 修改消息的转态
                System.out.println("confirm...correlationData[" + correlationData + "]==>ack[" + ack + "]==>[cause[" + cause + "]");
            }
        });
        // 设置消息没有抵达队列的失败回调
        rabbitTemplate.setReturnCallback(new RabbitTemplate.ReturnCallback() {
            /**
             * 只要消息没有投递给指定的队列，就触发这个失败回调
             * @param message 投递失败的消息详细信息
             * @param replyCode 回复的状态码
             * @param replyText 回复的文本内容
             * @param exchange 当时这个消息发给哪个交换机
             * @param routingKey 当时这个消息用哪个路由键
             */
            @Override
            public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
                //报错误了 。修改数据库当前消息的状态->错误。
                System.out.println("Fail Message[" + message + "]==>replyCode[" + replyCode + "]==>replyText[" + replyText + "]==>exchange[" + exchange + "]==>routingKey[" + routingKey + "]");
            }
        });
    }
}
